Cleaning up code (adding comments, removing unneeded lines, etc.)