Added helper function to set env vars