Switched to Random Forest + Feature Importance