Refactor code

just refactor