The test script mostly works