Changes to load_data and new method to add data