Updated test-1.py

Updated test-1.py