Added some tests for the buttons