added a couple tests