diff --git a/main.c b/main.c
index e504289..76ae01c 100644
--- a/main.c
+++ b/main.c
@@ -1,9 +1,8 @@
 #include <netinet/in.h>
 #include <arpa/inet.h>
-#include <stdio.h>
 #include <unistd.h>
-#include <errno.h>
-
+#include <stdlib.h>
+#include "easysock.h"
 void forward_data(int from_fd, int to_fd) {
 	int n = 0;
         char* buffer = malloc(3000*sizeof(char));