Add function for loading data for assignment 6