lisätty pari luokkaa ja tehty niiden logiikka kuntoon