Adding some rudimentary tests