Split load_data into more methods