Code for examples for the working memory tutorial