Update NoteGenMain.py