Adding Test Coverage