Added a class to test input