Add loop for inputting