Adapt to new treebeard API.
Merge pull request #4 from ddsc/byrman_fix-treebeard-issue

Adapt to new treebeard API.