small tweaks to button