Edit and save functionality added