Add tests for node
Update demo