Forgot test files. And adding a test.