fixed setup_data.py