update dependency
fix uniq token count when untrain
version bump to 0.2.2