updated the code style