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