Update from Kim

Modified the awesome code Wen wrote so that instead of strings, it
outputs a list of integers.