3D graphing in python