Class diagram control