Adding an initial unit test