Adding the class def of a graph node. Import it to use.
Removed main code.