Solved problem 4
Removed the iteration step to 1