Added schema for sampler