Add one function for data_io.py