Fixed bug in data set generation that took the wrong number of steps.