Trie code by Phython

class Trie:
	head = {}
    
    def add(self, word}:
    	cur = self.head
    
    	for char in word:
        	if char not in cur:
            	cur[char] = {}
            cur = cur[char]
        cur['*'] = True
    
    def search(self, word):
    	cur = self.head
        
        for char in word:
        	if char not in cur:
            	return False
            cur = cur[char]
            
        if '*' in cur:
        	return True
        else:
        	return False