IP: zigzag: Improve python bindings