add learn_vocab.py