Minor modification in merge sort