* added class diagram
* beautify class diagram