Fixed some types in class diagram.