Add support for specifying schemas and aggregation configs