Improve documentation of the Whitening trainer. This closes #152