Add setup_environment()