Adding tests.
Merge pull request #3 from fnothaft/tests

Adding tests.