Tests for task 1 fixed