Updated for Python API v2