Added the dictionary to count tokens