FIX 3d plots to use different shapes