Доделал refactoring