Text classification using python