Update main.c

D'oh!