Adding the python interface