Added python module
Now works with Python 3