Completed initial unit test