Add delegate callbacks for expand/collapse