Fixed task 4 issue.