refine code.