Необходимо добавить в существующий код новый функционал. Снизу приложил условие задания, справочную информацию. Исходный код программы вышлю исполнителю. Подробности в лс.
П.С. в справочной информации находятся плюс-минус те функции, с помощью которых необходимо реализовать функционал.