Fixed some errors in class diagram