Update tests'
Use functions instead of classes
Simplifying node constructors
Updating tests.