Fixes #11 Revert changes to button.