refactor

code refactor