Fixed node class (forgot __init__ and self)