Update to testing script for exercise 2