Add code to set attribute names in the environment