Add class to set base data sources