Fixing some of the code formatting that I messed up in the last commit