Cleaned up the code formatting a bit.