Add unittest code.