Подключил расчет волновых функций. Добавил выборку нужных и вывод в файл. Переделал потенциал под яму, ввел нужные переменные