Minor updates to file and main (aesthetics)