Update main.c

Inserted shutdown function.
In the future probably with a second button or something.