Added main class and annotations.
Moved demo code to test tree.