needs to add the table for feature importance