Adding code before removing it