Added UnitTest to tests