update to batch correction