Add a unit test for Ridge Regression