Step 3: Add Context