update python wrapper