Add protocol to references for cell and header