clean tests and add more tests for the button element