Remove math import