edited load_data.R