problem 4 forgot only one input