add a couple tests