Add initial unit test