Adding some quick tests