@xiangli-cmu Can you explain the change in this section just so I'm clear? My brain is fried right now. :)