New class diagram

Signed-off-by: Yuriy Rusinov <yrusinov@gmail.com>