Add full test for task 1