Add test for add