Added the 3d plot basics, though needs modification to work properly