Refine code style