Update driver to include type of data load