Almost got editing working