removed import of Heaviside function