Finishing touches to exercise 3