Add test for TrieNode class