Add a flag for whitening/not-whitening.