Add capability to set the layers.
Revert change for setting layers.