Simple quadratic equation solver, in Python