Removed bunch of old code in test.py