Update main.c

Fix control reaching end of non-void function