Better cognition, better school performance? Evidence from primary schools in China