Add function for loading test data.