refactor for STEP_SIZE variable