Removed unnecessary imports and slightly modified testing code
Merge pull request #6 from xDredzx/master

Removed unnecessary imports and slightly modified testing code