test.py changed to compare performance