Added initial unit test