Minor clean up (style)