Add unfold_head