UML diagram

There are a few places I am not sure. Might need someone to review it.