Used UI instead of Input and Output, deleted unused imports