Added the if that allows the loop to restart.