Minor changes in Test.py