Correcting solution of task_2