Text Classification Python Interview Question