Class diagram work