Add test for the button element