plot_gen.py not working when in wrong directory

The script `plot_gen.py` does not work if it is not run from inside the folder where it is located. Adding the following line after the imports resolves this:
```python
file_dir = os.path.realpath(os.path.dirname(__file__))
os.chdir(file_dir)
```