Cleaning up some of the code formatting