Update README with test.py