Added class diagram of model