added initial unit test