Minor changes in  test.py