Revert to digram model

The feature that I wanted to use did not work for some reason. Reverting back to the last known working state.