Cleaned up some code formatting