incorporating add_data.py code