Modified the way of finding the peak in the enhancement