Minor tweaks to prompt