Added code test coverage