Added vsm features and switched to random forest