corr_w_scatter: Allow corr to be given as subset of full matrix