refine code..