Updates in the test script

Signed-off-by: Herbert <herberthb12@gmail.com>