Some minor changes to prompt