Adding code to the equation