Small Change to Edit Function