Lisätty pääluokka ja päivitetty luokkakaavio