Add and Search Word - Data structure design python