Can AI improve your focus and memory?