Added UIInputBox and cleaned the code