Adding the loop in now