Drobne zmiany + 2 unit testy