tidy up code
tutorial 44