Step 4: Add assignments