Defined methods in button
Adding tests for TDD