Changes to test_basic.py