Worked on getting unit tests to work properly.