Small PR to fix a bug where setting display_margins=True in bode_plot() caused the title to disappear. The bug was caused by overwriting the ax variable...

Also added logic to allow the grid to be turned back on (it defaults to off if display_margins is True).

Unit tests included that catches the previous bug and verifies new functionality.