added MM's latest attempt of fixing the class diagram