Update main.c

This is the template.