fixing test script
fuller testing code