Close #2: Add code coverage