Add some features to README
Make some tests pass in python 3