add class for input handling