Some work on python wrapper