Move comment about TestInput to the header of class definition to make clear the purpose of the class