Modify model diagrams

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark: