3d plotting in matplotlib