Update python wrapper