Added a few tests, and comments.

I was not really sure where to start, so I done some simple things.
Avoid wildcard imports.