Now can be imported into python as a module