Fixed the input and output