modify DataSet to work with the gatech dataset