adding vanilla extract to cooled pudding
Yep, that should be fine. If you have trouble mixing it in thoroughly, try reheating the pudding in the microwave, a few seconds at a time, to make it thinner and more stirrable. (It’ll re-thicken when cooled.)