change  ML class diagram