Adds tests to cover button actions