Simplified test.py