Fix typo
fix typo
clean up code