Fixed typos
Simplified variable definition
Updated tests

- Matches new formulation of stateful linear regression