Add docstrings. Set 3D plot to white background.