Added Base model class