Lisasin kahele mudeli klassile ka testid aga vaja veel teste teha