Changed color scale to show optimization better