Setup unit tests for 2 classes