added second thing to test.py