Adding test coverage for other methods