Fixes to API diagram