Added node type, and implemented
Cleaned up comments in classification module
Added base cases