adding test.py file