On the mechanical side there were a bunch of 3rd party supplements for Dragon Heist that added some rules and whatnot that might help you here. The one I remember the name of is Here's to Crime, which, amongst other things, hacks the BitD flashback mechanic for 5e play.
Didn't play that supplement but I did play BitD and, indeed, it's a great game and the flashback mechanic does wonders.