Created test.py file