adding test.py