more updates to diagram