Add function arg() to theme environment.