We already added a `input_class` config that should do exactly what this is doing.