Add textFieldShouldReturn as well as redraw both views when button is clicked.