Refactering na iteratie 2