generator for graph

to be used to test the code
alternatively if we don't want to make matplotlib a requirement you could add something like:

```python
try:
    from matplotlib import pyplot as plt
except ImportError:
    print('matplotlibnot installed, please install it with "pip install matplotlib")
    exit()
```