Made changes to show OOV for training and test as well