making run_on_data.py work better