refactor(code): refactor code