minor changes of data_reader.py