Get 3D plotting working