Junit testing for add function