Updated function names in test script