Cleaned up some code and updated the styling for section