Which of the following cognitive abilities is primarily assessed by the Working Memory Index (WMI) on the Wechsler scales?