Finishing problem 2