Update to problem 3