Problem 5 changes

Update to problem 5.