update to tools for classification