Tweaking the diagram more