a little enhance
Merge pull request #1 from feitian124/master

use module and nicer variable name