how to make quadratic with python