Basic unit tests

As suggested in issue #3.