Class diagram update

Basic functionality