working classification using python