Update the python bindings to work with current version