diff --git a/project/Core/Inc/tcp_cmd.h b/project/Core/Inc/tcp_cmd.h
index d970a58..fa6f063 100644
--- a/project/Core/Inc/tcp_cmd.h
+++ b/project/Core/Inc/tcp_cmd.h
@@ -6,14 +6,19 @@
 
 #ifndef INC_TCP_CMD_H_
 #define INC_TCP_CMD_H_
-
-#include <stdio.h>
-#include <string.h>
-#include <tcp.h>
+#ifndef TESTING
 #include "lcd_api.h"
+#else
+#include "mocs.h"
+#endif
 #include "llfs.h"
 #include "log.h"
 
 void tcp_cmd_init(void);
 
+void tcp_cmd_init( void );
+void tcp_cmd_write(struct tcp_pcb* pcb, const char* str);
+void tcp_cmd_print_help(struct tcp_pcb* pcb);
+
+
 #endif /* INC_TCP_CMD_H_ */