minor changes in code style