More work on python side of the API