WIP: adding support for carry input/outputs and simplifying python API