1 byte fix replacing while with for loop in problem 2