Fix main.c