Add capability to set the layers.

Add capability to set the layers.