Added class for handling input