Tests

Write tests.