Added the button and made the whole thing a form