Added another unit test