Minor change

Adding class structure for all the futur node