Bug fix: forgot to import math library