updates to diagram