Add function for generating dicts for setup
Modify setup function for workloads config