Added import statement for math. 

I need to fix or comment out the Equation class; it's not done.