Improve test coverage and simplify notation
Add TODO
Add linear regression