Split the classes to separate files and upgrade prompt_toolkit.