modified reorganize_data.py