small changes in the python interface