Tweaks to button