доделал немного логики работы:)