updated to diagram