Edit UI and add submit button feature