Update UML diagram

Philip/Fredrik