Added UnitTest class