refactoring

Signed-off-by: SButterfly <sbutterfly@outlook.com>