Adding test coverage to branch