Improving a surrogate model for Jet Substructure Tuning: