Немного поправил названия классов