add 3d plotting