Python Code For Classification