Adding code injection