Переименовал классы и сделал мелкие фиксы