Add a couple tests, but one is failing