2d 3d shapes plotting in matplotlib