Tidy up
Tidy prior to demonstration 1