Increase test coverage by adding some tests