adds unit test for Unit Class