Matplotlib 3d Function