Added tests for button clicks