Add test for add-button