Adding python bindings