quadratic equation python code