Add some testing (minor)