Add check for edit content