refining code
refining code