Update data_handler.py