added node class and some stuff