Adding code updated