Move data generator stuff to own file