Added initial model class for tests purposes