Added rmse and tweaked feature selection