Add role functionality