Add the todo on the flatten function.