Finding the roots of a quadratic equation. (Python)