Доработал код