Tweaks to done button