Adding tests and button directive