fix typo
clean up code