Update sgen.py for NWChem