some little bug fix!
Merge pull request #61 from Kenny9999/master

some bad import fix and a little add on!