[ADD] change the button to button input