Доработал функционал