from Kth_largest import KthLargest def main(input_1, input_2): result = [] kth_largest_element = KthLargest(input_2[0][0], input_2[0][1]) result.append(None) for i in range(1, len(input_1)): if input_1[i] == "add": res = kth_largest_element.add(input_2[i][0]) result.append(res) return result