polished and refactored some styling for layout