Built out 'edit' and save functionality