updated input to have a class