cleaning up code more
tidying up main.c and adding a comment to Makefile