Update Class diagram