# Xvfb :99 -screen 0 1024x768x24 -ac &
# export DISPLAY=:99

python main_single_agent.py --model_config ./config/experiments/one_agent/gpt4o_expert.json --baseline one_agent --config ./config/puzzles.json