Re-design tree nodes. Write docs