Adding test.py to repo