Update classes and giagram