Update main.c

-update code for medium and high level difficulty