vib_gen: Check and correct formatting

Make sure all formatting is correct. See wiki and [this readme](https://github.com/devitocodes/devito_book/blob/master/fdm-devito-notebooks/README.md) for more info.

**Available at:** `fdm-devito-notebooks/01_vib/vib_gen.ipynb,` or [here](https://nbviewer.jupyter.org/github/devitocodes/devito_book/blob/master/fdm-devito-notebooks/01_vib/vib_gen.ipynb).

Linked with #88 