Add code reference wrapper