Tweaks to upgrade page