Add check for roles