Updated all-tokens to sort by proximity.

Added some implementation rough draft to the paper.