- Add unit test for node class
- Small tweaks to the node class