fix typo and add feature