Added data generator and tests for running it.