Add initialization for WordSet class