You can certainly use 'after' in sentence A. It is simply that native speakers more often use 'in' for future reference. And B will take either preposition easily.
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.