Cleaning up comments for main