Small change in get_corr_func