Removed commented bad line
Removed commented bad line
Added whitening through diag again, and entered some comments
Improved test function to use data and plot some stuff