Add answer to problem 1