Update main.c

Fixes #9