update class diagram docs