Modify dump_env.py to run the function