Linear regression only performs with 9 featuers (I guess that hardcoded from kaggle dataset, which also has 9 features)