test.py and README updated