modify input & output classes