Update to enable collation to be performed without variable importances
Minor fixes