Colourful diet linked to better memory