Removed dependency on math

Already imported NumPy, not necessary to import math as well.