Some enhancements for the document classifier