I added a while loop.