lisaa testeja + koodin siistimista