Change button.click() to submit()