Fixed r_sq calculation to match PLINK output