Update main.c

hopefully fixed main