update junit test for add