ENH: adding python module