Added a load data function to collect_data.py