Small mistake in comments fixed