Cleaned up test script.