Assortative marriage for cognitive abilities