refactor
improve comment