Adding test coverage for controllers