from trie import *
from trie_node import *

def main(input_1, input_2):
    result = []
    trie_obj = Trie()
    result.append(None)
    for i in range(1, len(input_1)):
        if input_1[i] == "insert":
            res = trie_obj.insert(input_2[i][0])
            result.append(None)
        elif input_1[i] == "search":
            res = trie_obj.search(input_2[i][0])
            result.append(res)
        elif input_1[i] == "search_prefix":
            res = trie_obj.search_prefix(input_2[i][0])
            result.append(res)

    return result