Edit functionality complete