Add logging for working memory size