Adding a unit test framework and updating tests