Added unittest.