Added anpohter line to test.py