Extend the following class definition so it overloads the input and output operators.

```
class Vect {
public:
Vect () {}
Vect ( double, double );
private:
double r; // magnitude
double theta; // direction (radians)
};
//
// Constructor 2: Initializes all components
//
Vect :: Vect( double vectMag, double vectDir )
{
r = vectMag;
theta = vectDir;
}
```

```
class Vect {
public:
Vect () {}
Vect ( double, double );
private:
double r; // magnitude
double theta; // direction (radians)
friend istream& operator>> ( istream&, Vect& );
friend ostream& operator<< ( ostream&, const Vect& );
};
//
// Constructor 2: Initializes all components
//
Vect :: Vect( double vectMag, double vectDir )
{
r = vectMag;
theta = vectDir;
}
//
// Input a Vect object
//
istream& operator>> ( istream& in, Vect& v )
{
in >> v.r >> v.theta;
return in;
}
//
// Display a Vect object
//
ostream& operator<< ( ostream& out, const Vect& v )
{
out << "radius = " << v.r << ", theta = " << v.theta;
return out;
}
```