This should be 
```python
import matplotlib.pyplot as plt
```