Edit and save functionality completed