Text Classification Python Interview Bit