Update data_viewer.py