Add tests to test breadth of functionality changes in assignment 2.