整形

refine my code.