Python quadratic formula script