added test.py file