Update test script for testing