Added config for regression stage