Add merge sort code