Class diagram changes based on added implementation