Add some rudimentary testing