MAINT: add code comment and make a small code change