Python wrapper updated