Text Classification Python Interview Coding Practice