Update for PR comments #115

Closed johnzhang27 closed 3 years ago