lisaa koodia, uusi pit testi ja uusi luokkakaavio lisatty