Add key for stream diagrams