Switch data generation script to Python 2. Update README.