Updated test-2.py

Updated test-2.py