incremental changes to run_nade.py