Small tidy up prob 4-9