Testien lisäys + koodin selkeytys