modified the flow a bit
added node tree base classes