Improve prompt_direction