Added unit-test (task 5-2)