prepare_data function

Prior data preparation