Minor changes to the styling