Cleaned up a bit of code formatting