Fix header
Cleaning up NDTree