Я добавил чтение из файла и упростил функцию main