Update main.c

I think this is right.