Replace conditional statement 'if max_prob_labels:' with indented assign_max_prob_label(), so that it will not return None