Finished UI diagram visualization. Fixed right-click selection.