Changed test to equal? and added another unit test