Updated main.c his might just work !