добавил функицю, поправил баг.