exercise 6 - include documenting and cleaning the code