Step 4: Embed Input Features