diff --git a/cv9/snake.c b/cv9/snake.c
index 3818cac..12f37e5 100644
--- a/cv9/snake.c
+++ b/cv9/snake.c
@@ -1,24 +1,6 @@
-#include <stdio.h>
+#include "snake.h"
 #include <stdlib.h>
 
-#define FOOD_COUNT 5
-
-struct snake {
-    int x;
-    int y;
-    struct snake* next;
-};
-
-struct state {
-    struct snake* snake;
-    int foodx[FOOD_COUNT];
-    int foody[FOOD_COUNT];
-    int sx;
-    int sy;
-    int width;
-    int height;
-};
-
 struct snake* add_snake(struct snake* snake, int x, int y) {
     struct snake* head = (struct snake*)malloc(sizeof(struct snake));
     if (head == NULL) {
@@ -57,8 +39,8 @@ void free_snake(struct snake* snake) {
     }
 }
 
-int main() {
-    // Place your testing code here
-    return 0;
+int step_state(struct state* state) {
+    // Implement step_state function here
+    return END_CONTINUE;
 }