Moon-Jumper Working Memory

I recently read this article on working memory and would like to incorporate some sort of working memory feature into Moon-Jumper, if the existing tools don't already cover it. Working memory helps with daily planning and thinking, but comes at the cost of distraction (story of my life), so I would like to pair a working memory module with a focus module, or have the functions be a single model.