Utils: Remove unused flatten func