Removed multiple save / load which was in as a test
Added code to plot objects