Исправлена диаграмма классов. Улучшенна логика и структура