tidy up main.c