Minor adjustement to style