clean the code
Merge pull request #5 from hkb1990/hkb

clean the code