FIX: fixes to feature importance