Model with working memory works a bit better