add a basic python wrapper