Cleaning up the main file