Add a couple tests