Update main.c

Fixed a cosmetic bug..