small changes to code style