Definitely want to return len(task_states), best_score, idx .  If a student has moved to a later step, it trumps a higher score with less steps.