ENH: integration into Python