Add test.py and update README.md