Add input classes for map reduce