Updating python API