Added test.py to repo