Refatcor code
minor change