Zadanie 2 with tests