Update class diagram and add task note