Add node types to the new layout