Added onSortComplete callback