Minor fixes to style