Add #!/usr/bin/python to to of data_gen.py.