Add more features to final model