Minor changes to diagram