Add tests for Button