from math import * import numpy as np