Add plot saving
Add 3d graph