class diagrams, small errors in code