Added function to set the environment layout