Add some of the tests for task_4