Поправил функцию