Add env var handling