Fixed bug with updating X*X' matrix

X*X' was always getting replaced with the current value before
updating, so old values weren't propagating
Updated tests for new version of StatefulLinearRegression