Adding in edit code