from math import *
from matplotlib.pyplot import *