Added modification to diagram