Add some quick tests so we don't break functionality