setting up to refactor the edit piece next