Add support for Python