Modified Tests.py