Add tests
Make root() and set() to return $this.
Add tests for set()