Corrected includes in main.c