Added a condition in prompt_yes_or_no() to use input or raw_input