In episode it never says to import math, thus an error message will arise for the math command. Similarly, when typing the original command import numpy as np, later it is still referenced as numpy, thus numpy must be imported alone in order for it to work.