refactoring code

Signed-off-by: YoungLeeForeverBoy <ly513782705@gmail.com>