Add rudimentary tests