Update class diagram, initial done