Update main.c

Function Updated