improve main input