Correcting mistake in solution of task_2