Add short term memory :D