Tests

For #4