Updated unit test comments