Text Classification Python Interviewbit