refactor the code style