Added model class for the viewer