Add basic unit tests. This closes #6