couple of minor changes for style