Refactored ShellUIPromptImpl a bit