Update main.c

made it simple