added info to data loading stage