add option to specify attribute key