Updated python API