Добавлена логика, устранена неисправность