First pass at unit tests

Per issue #61