Added hundred procent test coverage