Document Classification in Python