add step size parameter TODO parameter serach