incremental change to NADE.py