some code refactoring

Signed-off-by: anti-freak <antifreak2011@gmail.com>