Add nd::assign_na function