@xiangli-cmu I expanded on the function comment and a comment within the function itself.