prep_data function

------------------------------------------------------------------------------------ Prepare dataset