Creating test.py