Changed load_data to data and added figure plot