Add unittest