Some minor fixes to the styling