fixed not a module error
Merge pull request #1 from hellking4u/tests2

Tests2
completed test #1