Update TreeWalker.js