Simple Text Classification Python