Update for Python module