Add Pundit support