modefied main.c