whiten function

Fit and apply whitening in one call