Update main for input