fixing 3d plot