Modified main.c