Edited the main.c file