Cover remaining code with tests.