Update Test.py

Change code