UML Correction

Missed one thing in the main UML