Working on adding submit button functionality