done setting up test environment for problem2. Now write the tests.