Modify the program to allow any number of words to use as keys, ie: choose the size of your n-gram used in your chain