Add proper node class.