Add test coverage

Signed-off-by: Mark Thomas <markthethomas@gmail.com>