Updated test script for personal testing.