Import the math module in this code.

Make sure to pick import, followed by math.