vähän lisää koodia ja jotain testejä