Add initial unit tests
Add another unit test