Revise code style