Minor update

Fix comments