Add shared transitions from screenshot to markup

Fixes earlier issue where UI didn't get dismissed correctly.

Avoids removing the ExitTransitionCoordinator callback too early,
so that we get the hideSharedElements call.

Bug: 170138961
Bug: 159419046
Test: manual
Change-Id: I78193e46c97d01943df9b8438a29c6ee1e0106ff