Slight updates to style