Step 1: Prepare the Data