Similar to how groups in diagram as code get a title with some optional icon, it would be great if figures supported something similar. See image below.