add data loader code