Updated problem 5