Prettied up main.c