Added a unit test for feature scaling