Tests

As per #2.