Add input to Problem 1