Simple implementation on top of prompt_toolkit