updated test.py and readme