Fixed little bug in main.c