Improve Python wrapper API.