Updated problem 20