minor changes

 to neaten code