Add logic for display and hide submit button and form fields