Updated readme and test.py