Extend python bindings