[WIP][ENH] Add code coverage

As the title says.