improve class diagrams and documents