Added tests for task_5