Fix python 2 support
Add a bit more info in readme