To remain connected to Wi-Fi when shifting into gear, tap the Wi-Fi icon > Wi-Fi Settings > Remain Connected in Drive.