Ability to document within Model View (Data Diagram) tab.

Data models or diagrams can get really complex. It would benefit me to have the ability to categorize and label what it going on within my data model within Power Bi. That way I can quickly remember what I did to change data, or see how different data tables work with each other because of the documentation ability I have within the data model.

Specifically, I would like to have the ability to add text boxes and be able to add a zones to move my tables into if they are related to each other. And also an add shape function where I can also add a transparent fill color to it. These things would bring clarity to myself and others who use my data models.