Tidying code and figures

Update figures and remove all code and figures that was created during
development.