Initial Diagram highlighting implementation