Final submission

Tidy up the code, make it look like the mean example in the
coursework......