Modified main.c. Added screenshot