Adding the input and output class