refine code

refine